home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeViewKDE.py < prev    next >
Encoding:
Python Source  |  2009-04-27  |  35.1 KB  |  840 lines
  1. # DistUpgradeViewKDE.py 
  2. #  
  3. #  Copyright (c) 2007 Canonical Ltd
  4. #  
  5. #  Author: Jonathan Riddell <jriddell@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. from PyQt4.QtCore import *
  23. from PyQt4.QtGui import *
  24. from PyQt4 import uic
  25.  
  26. import sys
  27. import logging
  28. import time
  29. import subprocess
  30. import traceback
  31. import tempfile
  32.  
  33. import apt
  34. import apt_pkg
  35. import os
  36. import shutil
  37.  
  38. import pty
  39.  
  40. from DistUpgradeApport import *
  41.  
  42. from DistUpgradeController import DistUpgradeController
  43. from DistUpgradeView import DistUpgradeView, FuzzyTimeToStr, InstallProgress, FetchProgress
  44.  
  45. import select
  46. import gettext
  47. from DistUpgradeGettext import gettext as gett
  48.  
  49. def _(str):
  50.     return unicode(gett(str), 'UTF-8')
  51.  
  52. def utf8(str):
  53.   if isinstance(str, unicode):
  54.       return str
  55.   return unicode(str, 'UTF-8')
  56.  
  57. def loadUi(file, parent):
  58.     if os.path.exists(file):
  59.         uic.loadUi(file, parent)
  60.     else:
  61.         #FIXME find file
  62.         print "error, can't find file: " + file
  63.  
  64. class DumbTerminal(QTextEdit):
  65.     """ A very dumb terminal """
  66.     def __init__(self, installProgress, parent_frame):
  67.         " really dumb terminal with simple editing support "
  68.         QTextEdit.__init__(self, "", parent_frame)
  69.         self.installProgress = installProgress
  70.         self.setFontFamily("Monospace")
  71.         self.setFontPointSize(8)
  72.         self.setWordWrapMode(QTextOption.NoWrap)
  73.         self.setUndoRedoEnabled(False)
  74.         self.setOverwriteMode(True)
  75.         self._block = False
  76.         #self.connect(self, SIGNAL("cursorPositionChanged()"), 
  77.         #             self.onCursorPositionChanged)
  78.  
  79.     def fork(self):
  80.         """pty voodoo"""
  81.         (self.child_pid, self.installProgress.master_fd) = pty.fork()
  82.         if self.child_pid == 0:
  83.             os.environ["TERM"] = "dumb"
  84.         return self.child_pid
  85.  
  86.     def updateInterface(self):
  87.         (rlist, wlist, xlist) = select.select([self.installProgress.master_fd],[],[], 0)
  88.         if len(rlist) > 0:
  89.             line = os.read(self.installProgress.master_fd, 255)
  90.             self.insertWithTermCodes(utf8(line))
  91.         QApplication.processEvents()
  92.  
  93.     def insertWithTermCodes(self, text):
  94.         """ support basic terminal codes """
  95.         display_text = ""
  96.         for c in text:
  97.             # \b - backspace - this seems to comes as "^H" now ??!
  98.             if ord(c) == 8:
  99.                 self.insertPlainText(display_text)
  100.                 self.textCursor().deletePreviousChar()
  101.                 display_text=""
  102.             # \r - is filtered out
  103.             elif c == chr(13):
  104.                 pass
  105.             # \a - bell - ignore for now
  106.             elif c == chr(7):
  107.                 pass
  108.             else:
  109.                 display_text += c
  110.         self.insertPlainText(display_text)
  111.  
  112.     def keyPressEvent(self, ev):
  113.         """ send (ascii) key events to the pty """
  114.         # no master_fd yet
  115.         if not hasattr(self.installProgress, "master_fd"):
  116.             return
  117.         # special handling for backspace
  118.         if ev.key() == Qt.Key_Backspace:
  119.             #print "sent backspace"
  120.             os.write(self.installProgress.master_fd, chr(8))
  121.             return
  122.         # do nothing for events like "shift" 
  123.         if not ev.text():
  124.             return
  125.         # now sent the key event to the termianl as utf-8
  126.         os.write(self.installProgress.master_fd, ev.text().toUtf8())
  127.  
  128.     def onCursorPositionChanged(self):
  129.         """ helper that ensures that the cursor is always at the end """
  130.         if self._block:
  131.             return
  132.         # block signals so that we do not run into a recursion
  133.         self._block = True
  134.         self.moveCursor(QTextCursor.End)
  135.         self._block = False
  136.  
  137. class KDECdromProgressAdapter(apt.progress.CdromProgress):
  138.     """ Report the cdrom add progress """
  139.     def __init__(self, parent):
  140.         self.status = parent.window_main.label_status
  141.         self.progressbar = parent.window_main.progressbar_cache
  142.         self.parent = parent
  143.  
  144.     def update(self, text, step):
  145.         """ update is called regularly so that the gui can be redrawn """
  146.         if text:
  147.           self.status.setText(text)
  148.         self.progressbar.setValue(step/float(self.totalSteps))
  149.         QApplication.processEvents()
  150.  
  151.     def askCdromName(self):
  152.         return (False, "")
  153.  
  154.     def changeCdrom(self):
  155.         return False
  156.  
  157. class KDEOpProgress(apt.progress.OpProgress):
  158.   """ methods on the progress bar """
  159.   def __init__(self, progressbar, progressbar_label):
  160.       self.progressbar = progressbar
  161.       self.progressbar_label = progressbar_label
  162.       #self.progressbar.set_pulse_step(0.01)
  163.       #self.progressbar.pulse()
  164.  
  165.   def update(self, percent):
  166.       #if percent > 99:
  167.       #    self.progressbar.set_fraction(1)
  168.       #else:
  169.       #    self.progressbar.pulse()
  170.       #self.progressbar.set_fraction(percent/100.0)
  171.       self.progressbar.setValue(percent)
  172.       QApplication.processEvents()
  173.  
  174.   def done(self):
  175.       self.progressbar_label.setText("")
  176.  
  177. class KDEFetchProgressAdapter(FetchProgress):
  178.     """ methods for updating the progress bar while fetching packages """
  179.     # FIXME: we really should have some sort of "we are at step"
  180.     # xy in the gui
  181.     # FIXME2: we need to thing about mediaCheck here too
  182.     def __init__(self, parent):
  183.         FetchProgress.__init__(self)
  184.         # if this is set to false the download will cancel
  185.         self.status = parent.window_main.label_status
  186.         self.progress = parent.window_main.progressbar_cache
  187.         self.parent = parent
  188.  
  189.     def mediaChange(self, medium, drive):
  190.       msg = _("Please insert '%s' into the drive '%s'") % (medium,drive)
  191.       change = QMessageBox.question(self.parent.window_main, _("Media Change"), msg, QMessageBox.Ok, QMessageBox.Cancel)
  192.       if change == QMessageBox.Ok:
  193.         return True
  194.       return False
  195.  
  196.     def start(self):
  197.         #self.progress.show()
  198.         self.progress.setValue(0)
  199.         self.status.show()
  200.  
  201.     def stop(self):
  202.         self.parent.window_main.progress_text.setText("  ")
  203.         self.status.setText(_("Fetching is complete"))
  204.  
  205.     def pulse(self):
  206.         """ we don't have a mainloop in this application, we just call processEvents here and elsewhere"""
  207.         # FIXME: move the status_str and progress_str into python-apt
  208.         # (python-apt need i18n first for this)
  209.         FetchProgress.pulse(self)
  210.         self.progress.setValue(self.percent)
  211.         currentItem = self.currentItems + 1
  212.         if currentItem > self.totalItems:
  213.             currentItem = self.totalItems
  214.  
  215.         if self.currentCPS > 0:
  216.             self.status.setText(_("Fetching file %li of %li at %sB/s") % (currentItem, self.totalItems, apt_pkg.SizeToStr(self.currentCPS)))
  217.             self.parent.window_main.progress_text.setText("<i>" + _("About %s remaining") % unicode(FuzzyTimeToStr(self.eta), 'utf-8') + "</i>")
  218.         else:
  219.             self.status.setText(_("Fetching file %li of %li") % (currentItem, self.totalItems))
  220.             self.parent.window_main.progress_text.setText("  ")
  221.  
  222.         QApplication.processEvents()
  223.         return True
  224.  
  225. class KDEInstallProgressAdapter(InstallProgress):
  226.     """methods for updating the progress bar while installing packages"""
  227.     # timeout with no status change when the terminal is expanded
  228.     # automatically
  229.     TIMEOUT_TERMINAL_ACTIVITY = 240
  230.  
  231.     def __init__(self,parent):
  232.         InstallProgress.__init__(self)
  233.         self._cache = None
  234.         self.label_status = parent.window_main.label_status
  235.         self.progress = parent.window_main.progressbar_cache
  236.         self.progress_text = parent.window_main.progress_text
  237.         self.parent = parent
  238.         try:
  239.             self._terminal_log = open("/var/log/dist-upgrade/term.log","w")
  240.         except Exception, e:
  241.             # if something goes wrong (permission denied etc), use stdout
  242.             logging.error("Can not open terminal log: '%s'" % e)
  243.             self._terminal_log = sys.stdout
  244.         # some options for dpkg to make it die less easily
  245.         apt_pkg.Config.Set("DPkg::StopOnError","False")
  246.  
  247.     def startUpdate(self):
  248.         InstallProgress.startUpdate(self)
  249.         self.finished = False
  250.         # FIXME: add support for the timeout
  251.         # of the terminal (to display something useful then)
  252.         # -> longer term, move this code into python-apt 
  253.         self.label_status.setText(_("Applying changes"))
  254.         self.progress.setValue(0)
  255.         self.progress_text.setText(" ")
  256.         # do a bit of time-keeping
  257.         self.start_time = 0.0
  258.         self.time_ui = 0.0
  259.         self.last_activity = 0.0
  260.         self.parent.window_main.showTerminalButton.setEnabled(True)
  261.  
  262.     def error(self, pkg, errormsg):
  263.         InstallProgress.error(self, pkg, errormsg)
  264.         logging.error("got an error from dpkg for pkg: '%s': '%s'" % (pkg, errormsg))
  265.         # we do not report followup errors from earlier failures
  266.         if gettext.dgettext('dpkg', "dependency problems - leaving unconfigured") in errormsg:
  267.           return False
  268.         summary = _("Could not install '%s'") % pkg
  269.         msg = _("The upgrade will continue but the '%s' package may be "
  270.                 "in a not working state. Please consider submitting a "
  271.                 "bug report about it.") % pkg
  272.         msg = "<big><b>%s</b></big><br />%s" % (summary, msg)
  273.  
  274.         dialogue = QDialog(self.parent.window_main)
  275.         loadUi("dialog_error.ui", dialogue)
  276.         self.parent.translate_widget_children(dialogue)
  277.         dialogue.label_error.setText(utf8(msg))
  278.         if errormsg != None:
  279.             dialogue.textview_error.setText(utf8(errormsg))
  280.             dialogue.textview_error.show()
  281.         else:
  282.             dialogue.textview_error.hide()
  283.         dialogue.connect(dialogue.button_bugreport, SIGNAL("clicked()"), self.parent.reportBug)
  284.         dialogue.exec_()
  285.  
  286.     def conffile(self, current, new):
  287.         """ask question in case conffile has been changed by user"""
  288.         logging.debug("got a conffile-prompt from dpkg for file: '%s'" % current)
  289.         start = time.time()
  290.         prim = _("Replace the customized configuration file\n'%s'?") % current
  291.         sec = _("You will lose any changes you have made to this "
  292.                 "configuration file if you choose to replace it with "
  293.                 "a newer version.")
  294.         markup = "<span weight=\"bold\" size=\"larger\">%s </span> \n\n%s" % (prim, sec)
  295.         self.confDialogue = QDialog(self.parent.window_main)
  296.         loadUi("dialog_conffile.ui", self.confDialogue)
  297.         self.confDialogue.label_conffile.setText(markup)
  298.         self.confDialogue.textview_conffile.hide()
  299.         #FIXME, below to be tested
  300.         #self.confDialogue.resize(self.confDialogue.minimumSizeHint())
  301.         self.confDialogue.connect(self.confDialogue.show_difference_button, SIGNAL("clicked()"), self.showConffile)
  302.  
  303.         # now get the diff
  304.         if os.path.exists("/usr/bin/diff"):
  305.           cmd = ["/usr/bin/diff", "-u", current, new]
  306.           diff = utf8(subprocess.Popen(cmd, stdout=subprocess.PIPE).communicate()[0])
  307.           self.confDialogue.textview_conffile.setText(diff)
  308.         else:
  309.           self.confDialogue.textview_conffile.setText(_("The 'diff' command was not found"))
  310.         result = self.confDialogue.exec_()
  311.         self.time_ui += time.time() - start
  312.         # if replace, send this to the terminal
  313.         if result == QDialog.Accepted:
  314.             os.write(self.master_fd, "y\n")
  315.         else:
  316.             os.write(self.master_fd, "n\n")
  317.  
  318.     def showConffile(self):
  319.         if self.confDialogue.textview_conffile.isVisible():
  320.             self.confDialogue.textview_conffile.hide()
  321.             self.confDialogue.show_difference_button.setText(_("Show Difference >>>"))
  322.         else:
  323.             self.confDialogue.textview_conffile.show()
  324.             self.confDialogue.show_difference_button.setText(_("<<< Hide Difference"))
  325.  
  326.     def fork(self):
  327.         """pty voodoo"""
  328.         (self.child_pid, self.master_fd) = pty.fork()
  329.         if self.child_pid == 0:
  330.             os.environ["TERM"] = "dumb"
  331.             if (not os.environ.has_key("DEBIAN_FRONTEND") or
  332.                 os.environ["DEBIAN_FRONTEND"] == "kde"):
  333.                 os.environ["DEBIAN_FRONTEND"] = "noninteractive"
  334.             os.environ["APT_LISTCHANGES_FRONTEND"] = "none"
  335.         logging.debug(" fork pid is: %s" % self.child_pid)
  336.         return self.child_pid
  337.  
  338.     def statusChange(self, pkg, percent, status):
  339.         """update progress bar and label"""
  340.         # start the timer when the first package changes its status
  341.         if self.start_time == 0.0:
  342.           #print "setting start time to %s" % self.start_time
  343.           self.start_time = time.time()
  344.         self.progress.setValue(self.percent)
  345.         self.label_status.setText(unicode(status.strip(), 'UTF-8'))
  346.         # start showing when we gathered some data
  347.         if percent > 1.0:
  348.           self.last_activity = time.time()
  349.           self.activity_timeout_reported = False
  350.           delta = self.last_activity - self.start_time
  351.           # time wasted in conffile questions (or other ui activity)
  352.           delta -= self.time_ui
  353.           time_per_percent = (float(delta)/percent)
  354.           eta = (100.0 - self.percent) * time_per_percent
  355.           # only show if we have some sensible data (60sec < eta < 2days)
  356.           if eta > 61.0 and eta < (60*60*24*2):
  357.             self.progress_text.setText(_("About %s remaining") % FuzzyTimeToStr(eta))
  358.           else:
  359.             self.progress_text.setText(" ")
  360.  
  361.     def finishUpdate(self):
  362.         self.label_status.setText("")
  363.  
  364.     def updateInterface(self):
  365.         """
  366.         no mainloop in this application, just call processEvents lots here
  367.         it's also important to sleep for a minimum amount of time
  368.         """
  369.         # log the output of dpkg (on the master_fd) to the terminal log
  370.         while True:
  371.             try:
  372.                 (rlist, wlist, xlist) = select.select([self.master_fd],[],[], 0)
  373.                 if len(rlist) > 0:
  374.                     line = os.read(self.master_fd, 255)
  375.                     self._terminal_log.write(line)
  376.                     self.parent.terminal_text.insertWithTermCodes(utf8(line))
  377.                 else:
  378.                     break
  379.             except Exception, e:
  380.                 print e
  381.                 logging.debug("error reading from self.master_fd '%s'" % e)
  382.                 break
  383.  
  384.         # now update the GUI
  385.         try:
  386.           InstallProgress.updateInterface(self)
  387.         except ValueError, e:
  388.           logging.error("got ValueError from InstallProgress.updateInterface. Line was '%s' (%s)" % (self.read, e))
  389.           # reset self.read so that it can continue reading and does not loop
  390.           self.read = ""
  391.         # check about terminal activity
  392.         if self.last_activity > 0 and \
  393.            (self.last_activity + self.TIMEOUT_TERMINAL_ACTIVITY) < time.time():
  394.           if not self.activity_timeout_reported:
  395.             #FIXME bug 95465, I can't recreate this, so here's a hacky fix
  396.             try:
  397.                 logging.warning("no activity on terminal for %s seconds (%s)" % (self.TIMEOUT_TERMINAL_ACTIVITY, self.label_status.text()))
  398.             except UnicodeEncodeError:
  399.                 logging.warning("no activity on terminal for %s seconds" % (self.TIMEOUT_TERMINAL_ACTIVITY))
  400.             self.activity_timeout_reported = True
  401.           self.parent.window_main.konsole_frame.show()
  402.         QApplication.processEvents()
  403.         time.sleep(0.02)
  404.  
  405.     def waitChild(self):
  406.         while True:
  407.             self.updateInterface()
  408.             (pid, res) = os.waitpid(self.child_pid,os.WNOHANG)
  409.             if pid == self.child_pid:
  410.                 break
  411.         return os.WEXITSTATUS(res)
  412.  
  413. # inherit from the class created in window_main.ui
  414. # to add the handler for closing the window
  415. class UpgraderMainWindow(QWidget):
  416.  
  417.     def __init__(self):
  418.         QWidget.__init__(self)
  419.         #uic.loadUi("window_main.ui", self)
  420.         loadUi("window_main.ui", self)
  421.  
  422.     def setParent(self, parentRef):
  423.         self.parent = parentRef
  424.  
  425.     def closeEvent(self, event):
  426.         close = self.parent.on_window_main_delete_event()
  427.         if close:
  428.             event.accept()
  429.         else:
  430.             event.ignore()
  431.  
  432. class DistUpgradeViewKDE(DistUpgradeView):
  433.     """KDE frontend of the distUpgrade tool"""
  434.     def __init__(self, datadir=None, logdir=None):
  435.         # silence the PyQt4 logger
  436.         logger = logging.getLogger("PyQt4")
  437.         logger.setLevel(logging.INFO)
  438.         if not datadir:
  439.           localedir=os.path.join(os.getcwd(),"mo")
  440.         else:
  441.           localedir="/usr/share/locale/update-manager"
  442.  
  443.         # FIXME: i18n must be somewhere relative do this dir
  444.         try:
  445.           gettext.bindtextdomain("update-manager", localedir)
  446.           gettext.textdomain("update-manager")
  447.         except Exception, e:
  448.           logging.warning("Error setting locales (%s)" % e)
  449.  
  450.         #about = KAboutData("adept_manager","Upgrader","0.1","Dist Upgrade Tool for Kubuntu",KAboutData.License_GPL,"(c) 2007 Canonical Ltd",
  451.         #"http://wiki.kubuntu.org/KubuntuUpdateManager", "jriddell@ubuntu.com")
  452.         #about.addAuthor("Jonathan Riddell", None,"jriddell@ubuntu.com")
  453.         #about.addAuthor("Michael Vogt", None,"michael.vogt@ubuntu.com")
  454.         #KCmdLineArgs.init(["./dist-upgrade.py"],about)
  455.  
  456.         #self.app = KApplication()
  457.         self.app = QApplication(["update-manager"])
  458.  
  459.         if os.path.exists("/usr/share/icons/oxygen/48x48/apps/system-software-update.png"):
  460.             messageIcon = QPixmap("/usr/share/icons/oxygen/48x48/apps/system-software-update.png")
  461.         else:
  462.             messageIcon = QPixmap("/usr/share/icons/hicolor/48x48/apps/adept_manager.png")
  463.         self.app.setWindowIcon(QIcon(messageIcon))
  464.  
  465.         self.window_main = UpgraderMainWindow()
  466.         self.window_main.setParent(self)
  467.         self.window_main.show()
  468.  
  469.         self.prev_step = 0 # keep a record of the latest step
  470.  
  471.         self._opCacheProgress = KDEOpProgress(self.window_main.progressbar_cache, self.window_main.progress_text)
  472.         self._fetchProgress = KDEFetchProgressAdapter(self)
  473.         self._cdromProgress = KDECdromProgressAdapter(self)
  474.  
  475.         self._installProgress = KDEInstallProgressAdapter(self)
  476.  
  477.         # reasonable fault handler
  478.         sys.excepthook = self._handleException
  479.  
  480.         self.window_main.showTerminalButton.setEnabled(False)
  481.         self.app.connect(self.window_main.showTerminalButton, SIGNAL("clicked()"), self.showTerminal)
  482.  
  483.         #kdesu requires us to copy the xauthority file before it removes it when Adept is killed
  484.         copyXauth = tempfile.mktemp("", "adept")
  485.         if 'XAUTHORITY' in os.environ and os.environ['XAUTHORITY'] != copyXauth:
  486.             shutil.copy(os.environ['XAUTHORITY'], copyXauth)
  487.             os.environ["XAUTHORITY"] = copyXauth
  488.  
  489.         # Note that with kdesudo this needs --nonewdcop
  490.         ## create a new DCOP-Client:
  491.         #client = DCOPClient()
  492.         ## connect the client to the local DCOP-server:
  493.         #client.attach()
  494.  
  495.         #for qcstring_app in client.registeredApplications():
  496.         #    app = str(qcstring_app)
  497.         #    if app.startswith("adept"): 
  498.         #        adept = DCOPApp(qcstring_app, client)
  499.         #        adeptInterface = adept.object("MainApplication-Interface")
  500.         #        adeptInterface.quit()
  501.  
  502.         # This works just as well
  503.         subprocess.call(["killall", "adept_manager"])
  504.         subprocess.call(["killall", "adept_updater"])
  505.  
  506.         # init gettext
  507.         gettext.bindtextdomain("update-manager",localedir)
  508.         gettext.textdomain("update-manager")
  509.         self.translate_widget_children()
  510.         self.window_main.label_title.setText(self.window_main.label_title.text().replace("Ubuntu", "Kubuntu"))
  511.  
  512.         # setup terminal text in hidden by default spot
  513.         self.window_main.konsole_frame.hide()
  514.         self.konsole_frame_layout = QHBoxLayout(self.window_main.konsole_frame)
  515.         self.window_main.konsole_frame.setMinimumSize(600, 400)
  516.         self.terminal_text = DumbTerminal(self._installProgress, self.window_main.konsole_frame)
  517.         self.konsole_frame_layout.addWidget(self.terminal_text)
  518.         self.terminal_text.show()
  519.  
  520.         # for some reason we need to start the main loop to get everything displayed
  521.         # this app mostly works with processEvents but run main loop briefly to keep it happily displaying all widgets
  522.         QTimer.singleShot(10, self.exitMainLoop)
  523.         self.app.exec_()
  524.  
  525.     def exitMainLoop(self):
  526.         print "exitMainLoop"
  527.         self.app.exit()
  528.  
  529.     def translate_widget_children(self, parentWidget=None):
  530.         if parentWidget == None:
  531.             parentWidget = self.window_main
  532.         if isinstance(parentWidget, QDialog) or isinstance(parentWidget, QWidget):
  533.             if str(parentWidget.windowTitle()) == "Error":
  534.                 parentWidget.setWindowTitle( gettext.dgettext("kdelibs", "Error"))
  535.             else:
  536.                 parentWidget.setWindowTitle(_( str(parentWidget.windowTitle()) ))
  537.  
  538.         if parentWidget.children() != None:
  539.             for widget in parentWidget.children():
  540.                 self.translate_widget(widget)
  541.                 self.translate_widget_children(widget)
  542.  
  543.     def translate_widget(self, widget):
  544.         if isinstance(widget, QLabel) or isinstance(widget, QPushButton):
  545.             if str(widget.text()) == "&Cancel":
  546.                 widget.setText(unicode(gettext.dgettext("kdelibs", "&Cancel"), 'UTF-8'))
  547.             elif str(widget.text()) == "&Close":
  548.                 widget.setText(unicode(gettext.dgettext("kdelibs", "&Close"), 'UTF-8'))
  549.             elif str(widget.text()) != "":
  550.                 widget.setText( _(str(widget.text())).replace("_", "&") )
  551.  
  552.     def _handleException(self, exctype, excvalue, exctb):
  553.         """Crash handler."""
  554.  
  555.         if (issubclass(exctype, KeyboardInterrupt) or
  556.             issubclass(exctype, SystemExit)):
  557.             return
  558.  
  559.         # we handle the exception here, hand it to apport and run the
  560.         # apport gui manually after it because we kill u-m during the upgrade
  561.         # to prevent it from popping up for reboot notifications or FF restart
  562.         # notifications or somesuch
  563.         lines = traceback.format_exception(exctype, excvalue, exctb)
  564.         logging.error("not handled exception in KDE frontend:\n%s" % "\n".join(lines))
  565.         # we can't be sure that apport will run in the middle of a upgrade
  566.         # so we still show a error message here
  567.         apport_crash(exctype, excvalue, exctb)
  568.         if not run_apport():
  569.             tbtext = ''.join(traceback.format_exception(exctype, excvalue, exctb))
  570.             dialog = QDialog(self.window_main)
  571.             loadUi("dialog_error.ui", dialog)
  572.             self.translate_widget_children(self.dialog)
  573.             #FIXME make URL work
  574.             #dialog.connect(dialog.beastie_url, SIGNAL("leftClickedURL(const QString&)"), self.openURL)
  575.             dialog.crash_detail.setText(tbtext)
  576.             dialog.exec_()
  577.         sys.exit(1)
  578.  
  579.     def openURL(self, url):
  580.         """start konqueror"""
  581.         #need to run this else kdesu can't run Konqueror
  582.         #subprocess.call(['su', 'ubuntu', 'xhost', '+localhost'])
  583.         QDesktopServices.openUrl(QUrl(url))
  584.  
  585.     def reportBug(self):
  586.         """start konqueror"""
  587.         #need to run this else kdesu can't run Konqueror
  588.         #subprocess.call(['su', 'ubuntu', 'xhost', '+localhost'])
  589.         QDesktopServices.openUrl(QUrl("https://launchpad.net/ubuntu/+source/update-manager/+filebug"))
  590.  
  591.     def showTerminal(self):
  592.         if self.window_main.konsole_frame.isVisible():
  593.             self.window_main.konsole_frame.hide()
  594.             self.window_main.showTerminalButton.setText(_("Show Terminal >>>"))
  595.         else:
  596.             self.window_main.konsole_frame.show()
  597.             self.window_main.showTerminalButton.setText(_("<<< Hide Terminal"))
  598.         self.window_main.resize(self.window_main.sizeHint())
  599.  
  600.     def getFetchProgress(self):
  601.         return self._fetchProgress
  602.  
  603.     def getInstallProgress(self, cache):
  604.         self._installProgress._cache = cache
  605.         return self._installProgress
  606.  
  607.     def getOpCacheProgress(self):
  608.         return self._opCacheProgress
  609.  
  610.     def getCdromProgress(self):
  611.         return self._cdromProgress
  612.  
  613.     def updateStatus(self, msg):
  614.         self.window_main.label_status.setText(utf8(msg))
  615.  
  616.     def hideStep(self, step):
  617.         image = getattr(self.window_main,"image_step%i" % step)
  618.         label = getattr(self.window_main,"label_step%i" % step)
  619.         image.hide()
  620.         label.hide()
  621.  
  622.     def abort(self):
  623.         step = self.prev_step
  624.         if step > 0:
  625.             image = getattr(self.window_main,"image_step%i" % step)
  626.             if os.path.exists("/usr/share/icons/oxygen/16x16/actions/dialog-cancel.png"):
  627.                 cancelIcon = QPixmap("/usr/share/icons/oxygen/16x16/actions/dialog-cancel.png")
  628.             elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-cancel.png"):
  629.                 cancelIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-cancel.png")
  630.             else:
  631.                 cancelIcon = QPixmap("/usr/share/icons/crystalsvg/16x16/actions/cancel.png")
  632.             image.setPixmap(cancelIcon)
  633.             image.show()
  634.  
  635.     def setStep(self, step):
  636.         if os.path.exists("/usr/share/icons/oxygen/16x16/actions/dialog-ok.png"):
  637.             okIcon = QPixmap("/usr/share/icons/oxygen/16x16/actions/dialog-ok.png")
  638.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-ok.png"):
  639.             okIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-ok.png")
  640.         else:
  641.             okIcon = QPixmap("/usr/share/icons/crystalsvg/16x16/actions/ok.png")
  642.  
  643.         if os.path.exists("/usr/share/icons/oxygen/16x16/actions/arrow-right.png"):
  644.             arrowIcon = QPixmap("/usr/share/icons/oxygen/16x16/actions/arrow-right.png")
  645.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/16x16/actions/arrow-right.png"):
  646.             arrowIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/16x16/actions/arrow-right.png")
  647.         else:
  648.             arrowIcon = QPixmap("/usr/share/icons/crystalsvg/16x16/actions/1rightarrow.png")
  649.  
  650.         if self.prev_step:
  651.             image = getattr(self.window_main,"image_step%i" % self.prev_step)
  652.             label = getattr(self.window_main,"label_step%i" % self.prev_step)
  653.             image.setPixmap(okIcon)
  654.             image.show()
  655.             ##arrow.hide()
  656.         self.prev_step = step
  657.         # show the an arrow for the current step and make the label bold
  658.         image = getattr(self.window_main,"image_step%i" % step)
  659.         label = getattr(self.window_main,"label_step%i" % step)
  660.         image.setPixmap(arrowIcon)
  661.         image.show()
  662.         label.setText("<b>" + label.text() + "</b>")
  663.  
  664.     def information(self, summary, msg, extended_msg=None):
  665.         msg = "<big><b>%s</b></big><br />%s" % (summary,msg)
  666.  
  667.         dialogue = QDialog(self.window_main)
  668.         loadUi("dialog_error.ui", dialogue)
  669.         self.translate_widget_children(dialogue)
  670.         dialogue.label_error.setText(utf8(msg))
  671.         if extended_msg != None:
  672.             dialogue.textview_error.setText(utf8(extended_msg))
  673.             dialogue.textview_error.show()
  674.         else:
  675.             dialogue.textview_error.hide()
  676.         dialogue.button_bugreport.hide()
  677.         dialogue.setWindowTitle(_("Information"))
  678.  
  679.         if os.path.exists("/usr/share/icons/oxygen/48x48/status/dialog-information.png"):
  680.             messageIcon = QPixmap("/usr/share/icons/oxygen/48x48/status/dialog-information.png")
  681.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-information.png"):
  682.             messageIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-information.png")
  683.         else:
  684.             messageIcon = QPixmap("/usr/share/icons/crystalsvg/32x32/actions/messagebox_info.png")
  685.         dialogue.image.setPixmap(messageIcon)
  686.         dialogue.exec_()
  687.  
  688.     def error(self, summary, msg, extended_msg=None):
  689.         msg="<big><b>%s</b></big><br />%s" % (summary, msg)
  690.  
  691.         dialogue = QDialog(self.window_main)
  692.         loadUi("dialog_error.ui", dialogue)
  693.         self.translate_widget_children(dialogue)
  694.         dialogue.label_error.setText(utf8(msg))
  695.         if extended_msg != None:
  696.             dialogue.textview_error.setText(utf8(extended_msg))
  697.             dialogue.textview_error.show()
  698.         else:
  699.             dialogue.textview_error.hide()
  700.         dialogue.button_close.show()
  701.         self.app.connect(dialogue.button_bugreport, SIGNAL("clicked()"), self.reportBug)
  702.  
  703.         if os.path.exists("/usr/share/icons/oxygen/48x48/status/dialog-error.png"):
  704.             messageIcon = QPixmap("/usr/share/icons/oxygen/48x48/status/dialog-error.png")
  705.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-error.png"):
  706.             messageIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-error.png")
  707.         else:
  708.             messageIcon = QPixmap("/usr/share/icons/crystalsvg/32x32/actions/messagebox_critical.png")
  709.         dialogue.image.setPixmap(messageIcon)
  710.         dialogue.exec_()
  711.  
  712.         return False
  713.  
  714.     def confirmChanges(self, summary, changes, downloadSize, 
  715.                        actions=None, removal_bold=True):
  716.         """show the changes dialogue"""
  717.         # FIXME: add a whitelist here for packages that we expect to be
  718.         # removed (how to calc this automatically?)
  719.         DistUpgradeView.confirmChanges(self, summary, changes, downloadSize)
  720.         msg = unicode(self.confirmChangesMessage, 'UTF-8')
  721.         self.changesDialogue = QDialog(self.window_main)
  722.         loadUi("dialog_changes.ui", self.changesDialogue)
  723.  
  724.         self.changesDialogue.treeview_details.hide()
  725.         self.changesDialogue.connect(self.changesDialogue.show_details_button, SIGNAL("clicked()"), self.showChangesDialogueDetails)
  726.         self.translate_widget_children(self.changesDialogue)
  727.         self.changesDialogue.show_details_button.setText(_("Details") + " >>>")
  728.         self.changesDialogue.resize(self.changesDialogue.sizeHint())
  729.  
  730.         if os.path.exists("/usr/share/icons/oxygen/48x48/status/dialog-warning.png"):
  731.             warningIcon = QPixmap("/usr/share/icons/oxygen/48x48/status/dialog-warning.png")
  732.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-warning.png"):
  733.             warningIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-warning.png")
  734.         else:
  735.             warningIcon = QPixmap("/usr/share/icons/crystalsvg/32x32/actions/messagebox_warning.png")
  736.  
  737.         self.changesDialogue.question_pixmap.setPixmap(warningIcon)
  738.  
  739.         if actions != None:
  740.             cancel = actions[0].replace("_", "")
  741.             self.changesDialogue.button_cancel_changes.setText(cancel)
  742.             confirm = actions[1].replace("_", "")
  743.             self.changesDialogue.button_confirm_changes.setText(confirm)
  744.  
  745.         summaryText = unicode("<big><b>%s</b></big>" % summary, 'UTF-8')
  746.         self.changesDialogue.label_summary.setText(summaryText)
  747.         self.changesDialogue.label_changes.setText(msg)
  748.         # fill in the details
  749.         self.changesDialogue.treeview_details.clear()
  750.         self.changesDialogue.treeview_details.setHeaderLabels(["Packages"])
  751.         self.changesDialogue.treeview_details.header().hide()
  752.         for rm in self.toRemove:
  753.             self.changesDialogue.treeview_details.insertTopLevelItem(0, QTreeWidgetItem(self.changesDialogue.treeview_details, [_("Remove %s") % rm]) )
  754.         for inst in self.toInstall:
  755.             self.changesDialogue.treeview_details.insertTopLevelItem(0, QTreeWidgetItem(self.changesDialogue.treeview_details, [_("Install %s") % inst]) )
  756.         for up in self.toUpgrade:
  757.             self.changesDialogue.treeview_details.insertTopLevelItem(0, QTreeWidgetItem(self.changesDialogue.treeview_details, [_("Upgrade %s") % up]) )
  758.  
  759.         #FIXME resize label, stop it being shrinkable
  760.         res = self.changesDialogue.exec_()
  761.         if res == QDialog.Accepted:
  762.             return True
  763.         return False
  764.  
  765.     def showChangesDialogueDetails(self):
  766.         if self.changesDialogue.treeview_details.isVisible():
  767.             self.changesDialogue.treeview_details.hide()
  768.             self.changesDialogue.show_details_button.setText(_("Details") + " >>>")
  769.         else:
  770.             self.changesDialogue.treeview_details.show()
  771.             self.changesDialogue.show_details_button.setText("<<< " + _("Details"))
  772.         self.changesDialogue.resize(self.changesDialogue.sizeHint())
  773.  
  774.     def askYesNoQuestion(self, summary, msg, default='No'):
  775.         answer = QMessageBox.question(self.window_main, unicode(summary, 'UTF-8'), unicode("<font>") + unicode(msg, 'UTF-8'), QMessageBox.Yes|QMessageBox.No, QMessageBox.No)
  776.         if answer == QMessageBox.Yes:
  777.             return True
  778.         return False
  779.  
  780.     def confirmRestart(self):
  781.         messageBox = QMessageBox(QMessageBox.Question, _("Restart required"), _("<b><big>Restart the system to complete the upgrade</big></b>"), QMessageBox.NoButton, self.window_main)
  782.         yesButton = messageBox.addButton(QMessageBox.Yes)
  783.         noButton = messageBox.addButton(QMessageBox.No)
  784.         yesButton.setText(_("_Restart Now").replace("_", "&"))
  785.         noButton.setText(gettext.dgettext("kdelibs", "&Close"))
  786.         answer = messageBox.exec_()
  787.         if answer == QMessageBox.Yes:
  788.             return True
  789.         return False
  790.  
  791.     def processEvents(self):
  792.         QApplication.processEvents()
  793.  
  794.     def on_window_main_delete_event(self):
  795.         #FIXME make this user friendly
  796.         text = _("""<b><big>Cancel the running upgrade?</big></b>
  797.  
  798. The system could be in an unusable state if you cancel the upgrade. You are strongly advised to resume the upgrade.""")
  799.         text = text.replace("\n", "<br />")
  800.         cancel = QMessageBox.warning(self.window_main, _("Cancel Upgrade?"), text, QMessageBox.Yes, QMessageBox.No)
  801.         if cancel == QMessageBox.Yes:
  802.             return True
  803.         return False
  804.  
  805. if __name__ == "__main__":
  806.   
  807.   view = DistUpgradeViewKDE()
  808.   view.askYesNoQuestion("input box test","bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar ")
  809.  
  810.   if sys.argv[1] == "--test-term":
  811.       pid = view.terminal_text.fork()
  812.       if pid == 0:
  813.           subprocess.call(["bash"])
  814.           sys.exit()
  815.       while True:
  816.           view.terminal_text.updateInterface()
  817.           QApplication.processEvents()
  818.           time.sleep(0.01)
  819.  
  820.   if sys.argv[1] == "--show-in-terminal":
  821.       for c in open(sys.argv[2]).read():
  822.           view.terminal_text.insertWithTermCodes( c )
  823.           #print c, ord(c)
  824.           QApplication.processEvents()
  825.           time.sleep(0.05)
  826.       while True:
  827.           QApplication.processEvents()
  828.  
  829.   cache = apt.Cache()
  830.   for pkg in sys.argv[1:]:
  831.     if cache[pkg].isInstalled and not cache[pkg].isUpgradable: 
  832.       cache[pkg].markDelete(purge=True)
  833.     else:
  834.       cache[pkg].markInstall()
  835.   cache.commit(view._fetchProgress,view._installProgress)
  836.  
  837.   # keep the window open
  838.   while True:
  839.       QApplication.processEvents()
  840.